package fmppc.manager;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

import java.lang.invoke.SerializedLambda;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;

/**
 * 类描述:  属性Lambda处理工具
 *
 * @author 王雷
 * @version 1.0.0
 * @date 2022/12/9 0009 下午 7:33
 */
@Slf4j
public final class DataAttributeLambdaUtils {

    public static <T> String giveAttributeFields(AttributeFunction<T> fn) {
        final Field field = getField(fn);
        final TableId tableIdClass = field.getAnnotation(TableId.class);
        final TableField tableFieldClass = field.getAnnotation(TableField.class);
        if (tableIdClass != null && StringUtils.isNotEmpty(tableIdClass.value())) {
            return tableIdClass.value();
        } else if (tableFieldClass != null && StringUtils.isNotEmpty(tableFieldClass.value())) {
            return tableFieldClass.value();
        } else {
            return UnderlineToCamelUtils.camelToUnderline(field.getName());
        }
    }

    /**
     * 属性描述：属性头——IS
     *
     * @date : 2022/12/9 0009 下午 8:17
     */
    private static final String IS_ATTRIBUTE_TITLE = "is";

    /**
     * 属性描述：属性头——GET
     *
     * @date : 2022/12/9 0009 下午 8:17
     */
    private static final String GET_ATTRIBUTE_TITLE = "get";

    /**
     * 将bean的属性的get方法，作为lambda表达式传入时，获取get方法对应的属性Field
     *
     * @param fn  lambda表达式，bean的属性的get方法
     * @param <T> 泛型
     * @return 属性对象
     */
    private static <T> Field getField(AttributeFunction<T> fn) {
        // 从function取出序列化方法
        Method writeReplaceMethod;
        try {
            writeReplaceMethod = fn.getClass().getDeclaredMethod("writeReplace");
        } catch (NoSuchMethodException e) {
            throw new RuntimeException(e);
        }

        // 从序列化方法取出序列化的lambda信息
        boolean isAccessible = writeReplaceMethod.isAccessible();
        writeReplaceMethod.setAccessible(true);
        SerializedLambda serializedLambda;
        try {
            serializedLambda = (SerializedLambda) writeReplaceMethod.invoke(fn);
        } catch (IllegalAccessException | InvocationTargetException e) {
            throw new RuntimeException(e);
        }
        writeReplaceMethod.setAccessible(isAccessible);

        // 从lambda信息取出method、field、class等
        String implMethodName = serializedLambda.getImplMethodName();
        // 确保方法是符合规范的get方法，boolean类型是is开头
        if (!implMethodName.startsWith(IS_ATTRIBUTE_TITLE) && !implMethodName.startsWith(GET_ATTRIBUTE_TITLE)) {
            throw new RuntimeException("get方法名称: " + implMethodName + ", 不符合java bean规范");
        }

        // get方法开头为 is 或者 get，将方法名 去除is或者get，然后首字母小写，就是属性名
        int prefixLen = implMethodName.startsWith("is") ? 2 : 3;

        String fieldName = implMethodName.substring(prefixLen);
        String firstChar = fieldName.substring(0, 1);
        fieldName = fieldName.replaceFirst(firstChar, firstChar.toLowerCase());
        Field field;
        try {
            field = Class.forName(serializedLambda.getImplClass().replace("/", "."))
                    .getDeclaredField(fieldName);
        } catch (ClassNotFoundException | NoSuchFieldException e) {
            throw new RuntimeException(e);
        }

        return field;
    }
}
